# 1. 导包
import unittest
from config import BASE_DIR
import logging.handlers
from common.logging_use import init_log_config
from script.test_add_emp_params import TestAddEmpParams
from script.test_ihrm_login_params import TestIhrmLoginParams
from htmltestreport import HTMLTestReport

# 初始化日志 - 指定日志文件名、指定时间间隔单位、指定间隔数、日志文件备份数
init_log_config(BASE_DIR + "/log/ihrm.log")
# BASE_DIR+"/log/ihrm.log" 表示，将日志文件 ihrm.log 存放到 log/ 中

logging.info("开始生成测试报告")

# 2. 创建测试套件实例
suite = unittest.TestSuite()

# 3. 使用 实例组织测试类
suite.addTest(unittest.makeSuite(TestIhrmLoginParams))  # 【修改位置1】指定测试类，的类名
suite.addTest(unittest.makeSuite(TestAddEmpParams))  # 【修改位置2】指定测试类，的类名

# suite = unittest.defaultTestLoader.discover("./script", "test_*_params.py")

# 4. 使用 HTMLTestReport 生成 runner 对象
runner = HTMLTestReport("./report/ihrm.html",  # 【修改位置3】
                        description="登录接口和员工管理接口的测试报告",
                        title="接口测试v1.0")

# 5. 生成测试报告 run(suite)
runner.run(suite)

logging.info("测试报告生成 成功！")